mathematical function in python